package com.librelato.tela.datamodel;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.librelato.bean.CompradorBean;

public class CompradorDataModel extends ListDataModel<CompradorBean> implements SelectableDataModel<CompradorBean> {

	public CompradorDataModel(List<CompradorBean> data) {
		super(data);
	}
	@SuppressWarnings("unchecked")
	@Override
	public CompradorBean getRowData(String linha) {		
		List<CompradorBean> listaComp = (List<CompradorBean>) getWrappedData();
		for(CompradorBean comp : listaComp){
			if(comp.getCodigo().equals(linha))
				return comp;
			
		}
		
		return null;
	}

	@Override
	public Object getRowKey(CompradorBean linha) {
		return linha.getCodigo();
	}

}
